// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

#include <aries/parquet/xxhasher.h>

#define XXH_INLINE_ALL
#include <nebula/util/internal/xxhash.h>

namespace parquet {
    namespace {
        template<typename T>
        uint64_t XxHashHelper(T value, uint32_t seed) {
            return XXH64(reinterpret_cast<const void *>(&value), sizeof(T), seed);
        }

        template<typename T>
        void XxHashesHelper(const T *values, uint32_t seed, int num_values, uint64_t *results) {
            for (int i = 0; i < num_values; ++i) {
                results[i] = XxHashHelper(values[i], seed);
            }
        }
    } // namespace

    uint64_t XxHasher::Hash(int32_t value) const {
        return XxHashHelper(value, kParquetBloomXxHashSeed);
    }

    uint64_t XxHasher::Hash(int64_t value) const {
        return XxHashHelper(value, kParquetBloomXxHashSeed);
    }

    uint64_t XxHasher::Hash(float value) const {
        return XxHashHelper(value, kParquetBloomXxHashSeed);
    }

    uint64_t XxHasher::Hash(double value) const {
        return XxHashHelper(value, kParquetBloomXxHashSeed);
    }

    uint64_t XxHasher::Hash(const FLBA *value, uint32_t len) const {
        return XXH64(reinterpret_cast<const void *>(value->ptr), len, kParquetBloomXxHashSeed);
    }

    uint64_t XxHasher::Hash(const Int96 *value) const {
        return XXH64(reinterpret_cast<const void *>(value->value), sizeof(value->value),
                     kParquetBloomXxHashSeed);
    }

    uint64_t XxHasher::Hash(const ByteArray *value) const {
        return XXH64(reinterpret_cast<const void *>(value->ptr), value->len,
                     kParquetBloomXxHashSeed);
    }

    void XxHasher::Hashes(const int32_t *values, int num_values, uint64_t *hashes) const {
        XxHashesHelper(values, kParquetBloomXxHashSeed, num_values, hashes);
    }

    void XxHasher::Hashes(const int64_t *values, int num_values, uint64_t *hashes) const {
        XxHashesHelper(values, kParquetBloomXxHashSeed, num_values, hashes);
    }

    void XxHasher::Hashes(const float *values, int num_values, uint64_t *hashes) const {
        XxHashesHelper(values, kParquetBloomXxHashSeed, num_values, hashes);
    }

    void XxHasher::Hashes(const double *values, int num_values, uint64_t *hashes) const {
        XxHashesHelper(values, kParquetBloomXxHashSeed, num_values, hashes);
    }

    void XxHasher::Hashes(const Int96 *values, int num_values, uint64_t *hashes) const {
        for (int i = 0; i < num_values; ++i) {
            hashes[i] = XXH64(reinterpret_cast<const void *>(values[i].value),
                              sizeof(values[i].value), kParquetBloomXxHashSeed);
        }
    }

    void XxHasher::Hashes(const ByteArray *values, int num_values, uint64_t *hashes) const {
        for (int i = 0; i < num_values; ++i) {
            hashes[i] = XXH64(reinterpret_cast<const void *>(values[i].ptr), values[i].len,
                              kParquetBloomXxHashSeed);
        }
    }

    void XxHasher::Hashes(const FLBA *values, uint32_t type_len, int num_values,
                          uint64_t *hashes) const {
        for (int i = 0; i < num_values; ++i) {
            hashes[i] = XXH64(reinterpret_cast<const void *>(values[i].ptr), type_len,
                              kParquetBloomXxHashSeed);
        }
    }
} // namespace parquet
